tensor([[[0.3300, 0.3300],
         [0.3300, 0.3300]],

        [[0.3800, 0.3800],
         [0.3800, 0.3800]],

        [[0.2900, 0.2900],
         [0.2900, 0.2900]]], grad_fn=<RoundBackward1>)